#!/bin/bash
# 佛山大学知识问答系统 - 模型下载脚本

# 激活虚拟环境
source .venv/bin/activate

# 创建模型目录
mkdir -p models

echo "开始下载模型..."

# 1. 下载Qwen3-4B主模型 (约8GB)
echo "正在下载 Qwen3-4B-Instruct-2507 模型..."
huggingface-cli download Qwen/Qwen3-4B-Instruct-2507 \
    --local-dir "./models/Qwen3-4B-Instruct-2507" \
    --local-dir-use-symlinks False

# 2. 下载中文embedding模型 (约400MB)
echo "正在下载 m3e-base embedding模型..."
huggingface-cli download moka-ai/m3e-base \
    --local-dir "./models/m3e-base" \
    --local-dir-use-symlinks False

echo "模型下载完成！"
echo "模型位置:"
echo "- Qwen3-4B: ./models/Qwen3-4B-Instruct-2507"
echo "- m3e-base: ./models/m3e-base"